# -*- coding: UTF-8 -*-

from loader import *
from fullynn import Network as fn
from ann import network as an

test_datas , test_labels = get_test_data_set_x()
train_datas , train_labels = get_training_data_set_x()

fnn = fn.Network([28*28,300,10])

#ann = an.Network([28*28,300,10])
fnn.train(train_labels,train_datas,0.01,10)

print fnn.predict(test_datas[0])